# 使用官方的 PyTorch 镜像作为基础镜像
FROM pytorch/pytorch:latest

# 设置工作目录
WORKDIR /app

# 更新包管理器并安装基本依赖
RUN apt-get update && apt-get install -y \
    python3-pip \
    python3-dev \
    libcuda1-450 \
    && rm -rf /var/lib/apt/lists/*

# 安装 Python 依赖
RUN pip install --no-cache-dir \
    torch \
    transformers \
    numpy \
    pandas \
    scikit-learn

RUN pip install -r requirements.txt
RUN pip install pyqlib

# 复制你的模型和代码到容器中
COPY . /app

# 启动容器时运行模型脚本
CMD ["python3", "examples/prediction_example.py"]
